# -*- coding: utf-8 -*-
"""
Created on Tue Apr 23 11:51:28 2024

@author: 29865
"""
import matplotlib.pyplot as plt
import numpy as np

fig = plt.figure()              # 创建画布

ax1 = plt.subplot(211)          # 分为2×1图形阵，选择第1张图片绘图
ax1.set(title='y=x^2 & y=x')    # 添加标题

ax1.set(xlabel='x')             # 添加x轴名称‘x’
ax1.set(ylabel='y')             # 添加y轴名称‘y’
ax1.set(xlim=(0, 1))            # 指定x轴范围（0,1）
ax1.set(ylim=(0, 1))            # 指定y轴范围（0,1）
ax1.set(xticks=[0, 0.3, 0.6, 1])# 设置x轴刻度
ax1.set(yticks=[0, 0.5, 1])     # 设置y轴刻度

x = np.linspace(0, 1, 1001)     # 生成数组，0-1，平均分为1000份
ax1.plot(x, x ** 2)
ax1.plot(x, x)
ax1.legend(['y=x^2', 'y=x'])    # 添加图例
# ax1.savefig('1.png')          # 保存图片